cd ../..

gpus=$1

# python joint_main.py \
#     --gpus ${gpus}, \
#     --opt_conf ./configs/opt_joint.yaml \
#     --exp_name joint/BYOLA_OOD_itpl_cmpst \
#     --grad_compensate True \
#    --joint_itpl_exponent 0 \
#    --joint_itpl_warmup 0
for exp in 20 80 640
do 
    python joint_main.py \
        --gpus ${gpus}, \
        --opt_conf ./configs/opt_joint.yaml \
        --exp_name joint/BYOLA_OOD_CRNN_itpl_wo_mt \
        --feat_dim 256 \
        --grad_compensate False \
        --joint_itpl_exponent ${exp} \
        --joint_itpl_warmup 80
done    # --resume_from_checkpoint /home/shaonian/SED/sssl_sed/codes/exp/joint/BYOLA_OOD_CRNN_itpl/version_0/last.ckpt